自定义中间件

@app.middleware("http")
async def add_process_time_header(request: Request, call_next):
      # 请求前
    start_time = time.time()
    response = await call_next(request)
    # 响应后
    process_time = time.time() - start_time
    response.headers["X-Process-Time"] = str(process_time)
    print(process_time)
    return response

fastapi内置中间件

除了自定义中间件,我们也可以直接使用其他的中间件。

FastAPI通过 app.add_middleware() 操作来引入已定义的中间件。

from fastapi import FastAPI
from unicorn import UnicornMiddleware

app = FastAPI()

app.add_middleware(UnicornMiddleware, some_config="rainbow")

*app.add_middleware()*第一个参数是中间件类,同时还支持中间件的其他附加信息。

FastAPI提供了一些常用的中间件。

一、HTTPSRedirectMiddleware

HTTPSRedirectMiddleware强制发来的请求协议必须是https或者wss。
from fastapi import FastAPI
from fastapi.middleware.httpsredirect import HTTPSRedirectMiddleware

app = FastAPI()

app.add_middleware(HTTPSRedirectMiddleware)


@app.get("/")
async def main():
    return {"message": "Hello World"}

二、TrustedHostMiddleware

TrustedHostMiddleware强制发来的请求必须在Header信息中设置了Host选项,为了避免HTTP Host Header攻击。
from fastapi import FastAPI
from fastapi.middleware.trustedhost import TrustedHostMiddleware

app = FastAPI()

app.add_middleware(
    TrustedHostMiddleware, allowed_hosts=["example.com", "*.example.com"]
)


@app.get("/")
async def main():
    return {"message": "Hello World"}

三、GZipMiddleware

当请求的头信息 Accept-Encoding 字段带有”gzip”时,GZipMiddleware负责完成相应的返回结果处理。

GZipMiddleware 支持 *`minimum_size`* 参数:当返回结果大小小于指定值时不启用压缩。(单位为字节,默认值为500)
from fastapi import FastAPI
from fastapi.middleware.gzip import GZipMiddleware

app = FastAPI()

app.add_middleware(GZipMiddleware, minimum_size=1000)


@app.get("/")
async def main():
    return "somebigcontent"

参考资料:https://www.starlette.io/middleware/

cors中间件参考资料:https://www.cnblogs.com/mazhiyong/p/12987619.html